from selenium import webdriver
from tests import ChatLogin  # 确保导入 ChatLogin 类

class TestSuite:
    def __init__(self):
        # 启动浏览器
        self.driver = webdriver.Chrome()
        self.chat_login = ChatLogin.ChatLogin()  # 初始化 ChatLogin 实例

    def run_tests(self):
        # 运行所有测试用例
        self.chat_login.LoginFailTest()  # 调用登录失败测试
        self.chat_login.LoginSucTest()    # 调用登录成功测试
        # 添加其他测试用例...

    def cleanup(self):
        # 关闭浏览器
        self.driver.quit()

if __name__ == "__main__":
    suite = TestSuite()
    try:
        suite.run_tests()  # 执行测试用例
    finally:
        suite.cleanup()  # 确保浏览器在测试后关闭
